/*
 * This file is part of the eccOS Library.
 * Copyright (C) 2007.
 *
 * Version: MPL 1.1/GPL 2.0/LGPL 2.1
 *
 * The contents of this file are subject to the Mozilla Public License Version
 * 1.1 (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 * http://www.mozilla.org/MPL/
 *
 * Software distributed under the License is distributed on an "AS IS" basis,
 * WITHOUT WARRANTY OF ANY KIND, either express or implied. See the License
 * for the specific language governing rights and limitations under the
 * License.
 *
 * Alternatively, the contents of this file may be used under the terms of
 * either the GNU General Public License Version 2 or later (the "GPL"), or
 * the GNU Lesser General Public License Version 2.1 or later (the "LGPL"),
 * in which case the provisions of the GPL or the LGPL are applicable instead
 * of those above. If you wish to allow use of your version of this file only
 * under the terms of either the GPL or the LGPL, and not to allow others to
 * use your version of this file under the terms of the MPL, indicate your
 * decision by deleting the provisions above and replace them with the notice
 * and other provisions required by the GPL or the LGPL. If you do not delete
 * the provisions above, a recipient may use your version of this file under
 * the terms of any one of the MPL, the GPL or the LGPL.
 */
package org.openeccos;

import java.io.IOException;
import java.util.Hashtable;

import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class DownloadServlet extends HttpServlet {

	private static Hashtable<String, Document> documents = new Hashtable<String, Document>();

	protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
		final String docId = request.getParameter("docId");
		if (docId == null) {
			response.sendError(HttpServletResponse.SC_BAD_REQUEST);
		} else {
			Document doc = getDocument(docId);
			if (doc == null) {
				response.sendError(HttpServletResponse.SC_BAD_REQUEST);
			} else {
				response.setHeader("Content-Disposition", "inline; filename=\"" + doc.getName() + "\"");
				response.setContentType(doc.getMimeType());
				response.setContentLength(doc.getContentLength());
				response.getOutputStream().write(doc.getContents());
			}
			
			//delete from cache after 1 minute
			Thread t = new Thread() {
				public void run() {
					try {
						Thread.sleep(60 * 1000);
					} catch (InterruptedException e) {
						e.printStackTrace();
					}
					documents.remove(docId);
					//System.out.println("Doc " + docId + " removed. Docs in cache: " + documents.size());
				}
			};
			t.start();
		}
	}

	public static void addDocument(Document doc) {
		documents.put(doc.getId(), doc);
	}

	private static Document getDocument(String id) {
		return documents.get(id);
	}

	public static class Document {

		private String id;
		private String name;
		private String mimeType;
		private int contentLength;
		private byte[] contents;

		public int getContentLength() {
			return contentLength;
		}

		public void setContentLength(int contentLength) {
			this.contentLength = contentLength;
		}

		public byte[] getContents() {
			return contents;
		}

		public void setContents(byte[] contents) {
			this.contents = contents;
		}

		public String getMimeType() {
			return mimeType;
		}

		public void setMimeType(String mimeType) {
			this.mimeType = mimeType;
		}

		public String getName() {
			return name;
		}

		public void setName(String name) {
			this.name = name;
		}

		public String getId() {
			return id;
		}

		public void setId(String id) {
			this.id = id;
		}
	}
}